tensor([[[[[0.7200, 0.0000],
           [0.0000, 0.7200]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]]],



        [[[[0.0000, 1.0000],
           [1.0000, 0.0000]],

          [[0.0500, 0.0000],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.0000],
           [0.5500, 0.0000]]]],



        [[[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.9500, 1.0000],
           [1.0000, 1.0000]]],


         [[[1.0000, 1.0000],
           [1.0000, 1.0000]],

          [[1.0000, 1.0000],
           [0.1900, 1.0000]]]],



        [[[[0.2800, 0.0000],
           [0.0000, 0.2800]],

          [[0.0000, 0.0000],
           [0.0000, 0.0000]]],


         [[[0.0000, 0.0000],
           [0.0000, 0.0000]],

          [[0.0000, 0.0000],
           [0.2600, 0.0000]]]]], grad_fn=<RoundBackward1>)